package com.tenbent.product.base.security.impl;

import com.tenbent.product.base.security.CustomAccessDecisionManager;
import org.apache.commons.collections.CollectionUtils;
import org.springframework.security.access.AccessDeniedException;
import org.springframework.security.access.ConfigAttribute;
import org.springframework.security.authentication.InsufficientAuthenticationException;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.GrantedAuthority;
import org.springframework.stereotype.Component;

import java.util.Collection;
import java.util.Iterator;

/**
 * 自定义权限判断器
 *
 * 这里遍历判断该url所需的角色看用户是否具备，有具备则返回，都不具备则抛出AccessDeniedException异常
 *
 * authentication 登录用户
 *
 * o url
 *
 * configAttributes 资源对应的所有角色权限
 *
 * @author Randy
 *
 *         Created by ThinkPad on 2017/9/7.
 */
public class CustomAccessDecisionManagerImpl implements CustomAccessDecisionManager {

	@Override
	public void decide(Authentication authentication, Object o, Collection<ConfigAttribute> configAttributes)
			throws AccessDeniedException, InsufficientAuthenticationException {

		if (CollectionUtils.isEmpty(configAttributes)) {
			// 此处可以设置成没有配置的url 可以认为是全部同行，也可以认为是没有权限，需要根据具体场景决定。
			// return; //认为没有配置的url全部放行。
			throw new AccessDeniedException("Access Denied"); // 认为没有配置的url全部没有权限。
		}
		Iterator<ConfigAttribute> ite = configAttributes.iterator();
		while (ite.hasNext()) {
			ConfigAttribute ca = ite.next();
			String needRole = ca.getAttribute();
			// 判断登录用用户的权限是否具有访问资源(url)的权限
			for (GrantedAuthority ga : authentication.getAuthorities()) {
				if (ga.getAuthority().trim().equals(needRole.trim())) {
					// 匹配到有对应角色,则允许通过
					return;
				}
			}
		}
		// 该url有配置权限,但是当然登录用户没有匹配到对应权限,则禁止访问
		throw new AccessDeniedException("Access Denied");
	}

	@Override
	public boolean supports(ConfigAttribute configAttribute) {
		return true;
	}

	@Override
	public boolean supports(Class<?> aClass) {
		return true;
	}
}
